tensor([[0.4300, 0.4300, 0.4300, 0.4300, 0.4300],
        [0.5700, 0.5700, 0.5700, 0.5700, 0.5700]], grad_fn=<RoundBackward1>)